package mybatis.auto.createpojo;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class MybatisGeneratorUtil {

    public static void main(String[] args) {
        try {
            System.out.println("start generator ...");
            List<String> warnings = new ArrayList<String>();
            boolean overwrite = true;

            // generatorMySQL.xml文件的位置
            File configFile = new File("D:\\generator\\src\\main\\java\\mybatis\\auto\\createpojo\\generatorMySQL.xml");
            ConfigurationParser cp = new ConfigurationParser(warnings);
            Configuration config = cp.parseConfiguration(configFile);
            DefaultShellCallback callback = new DefaultShellCallback(overwrite);
            MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);

            myBatisGenerator.generate(null);
           /* List<Context> lc = config.getContexts();
            for(Context c:lc){
                List<TableConfiguration> list = c.getTableConfigurations();
                for(TableConfiguration tc:list){
                    String s = tc.getTableName();
                    System.out.println(s);
                }
            }*/
           /* List<GeneratedJavaFile> list = myBatisGenerator.getGeneratedJavaFiles();
            for(GeneratedJavaFile gjf:list){
                String s = gjf.getTargetPackage();
                System.out.println(s);
            }*/
            System.out.println("end generator!");
        } catch (IOException e) {
            e.printStackTrace();
        } catch (XMLParserException e) {
            e.printStackTrace();
        } catch (InvalidConfigurationException e) {
            e.printStackTrace();
        } catch (SQLException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}
